package uscdb.util;

import org.apache.log4j.Logger;

import java.io.File;
import java.io.FileOutputStream;

import com.github.junrar.Archive;
import com.github.junrar.rarfile.FileHeader;

public class RarUtil {
	/**
	 * Logger for this class
	 */
	private static final Logger logger = Logger.getLogger(RarUtil.class);

	/*** 这里用到了synchronized ，也就是防止出现并发问题 ***/
	/**
	 * 功能：解压rar文件
	 * 
	 * @param tarFileName
	 *            原始rar路径
	 * @param extPlace
	 *            解压到的文件夹
	 * @throws Exception
	 */
	public static synchronized Boolean untar(String tarFileName, String extPlace) {
		return unRarFile(tarFileName, extPlace);
	}

	/**
	 * 
	 * @param sourceFile
	 *            原始的rar File 对象
	 * @param extPlace
	 *            解压到的文件夹
	 * @return
	 */
	public static synchronized Boolean untar(File sourceFile, String extPlace) {
		return unRarFile(sourceFile, extPlace);
	}

	/**
	 * 功能：根据File对象解压文件
	 * 
	 * @param sourceFile
	 *            原始的rar File 对象
	 * @param dstDirectoryPath
	 *            目标的解压路径
	 * @return
	 */
	private static Boolean unRarFile(File sourceFile, String dstDirectoryPath) {
		Boolean result = false;
		File dstDiretory = new File(dstDirectoryPath);
		if (!dstDiretory.exists()) {// 目标目录不存在时，创建该文件夹
			dstDiretory.mkdirs();
		}
		Archive a = null;
		try {
			a = new Archive(sourceFile);
			if (a != null) {
				// a.getMainHeader().print(); // 打印文件信息.
				FileHeader fh = a.nextFileHeader();
				while (fh != null) {
					// 防止文件名中文乱码问题的处理
					String fileName = fh.getFileNameW().isEmpty() ? fh.getFileNameString() : fh.getFileNameW();
					if (fh.isDirectory()) { // 文件夹
						File fol = new File(dstDirectoryPath + File.separator + fileName);
						fol.mkdirs();
					} else { // 文件
						File out = new File(dstDirectoryPath + File.separator + fileName.trim());
						try {
							if (!out.exists()) {
								if (!out.getParentFile().exists()) {// 相对路径可能多级，可能需要创建父目录.
									out.getParentFile().mkdirs();
								}
								out.createNewFile();
							}
							FileOutputStream os = new FileOutputStream(out);
							a.extractFile(fh, os);
							os.close();
						} catch (Exception ex) {
							logger.error(ex.getMessage(), ex);
						}
					}
					fh = a.nextFileHeader();
				}
				a.close();
				result = true;
			}
		} catch (Exception e) {
			logger.error(e.getMessage(), e);
		}
		return result;
	}

	/**
	 * 根据原始rar路径，解压到指定文件夹下.
	 * 
	 * @param srcRarPath
	 *            原始rar路径
	 * @param dstDirectoryPath
	 *            解压到的文件夹
	 */
	public static Boolean unRarFile(String srcRarPath, String dstDirectoryPath) {
		Boolean result = false;
		if (!srcRarPath.toLowerCase().endsWith(".rar")) {
			// System.out.println("非rar文件！");
			logger.error("非rar文件！");
			return result;
		}
		File dstDiretory = new File(dstDirectoryPath);
		if (!dstDiretory.exists()) {// 目标目录不存在时，创建该文件夹
			dstDiretory.mkdirs();
		}
		Archive a = null;
		try {
			a = new Archive(new File(srcRarPath));
			if (a != null) {
				// a.getMainHeader().print(); // 打印文件信息.
				FileHeader fh = a.nextFileHeader();
				while (fh != null) {
					// 防止文件名中文乱码问题的处理
					String fileName = fh.getFileNameW().isEmpty() ? fh.getFileNameString() : fh.getFileNameW();
					if (fh.isDirectory()) { // 文件夹
						File fol = new File(dstDirectoryPath + File.separator + fileName);
						fol.mkdirs();
					} else { // 文件
						File out = new File(dstDirectoryPath + File.separator + fileName.trim());
						try {
							if (!out.exists()) {
								if (!out.getParentFile().exists()) {// 相对路径可能多级，可能需要创建父目录.
									out.getParentFile().mkdirs();
								}
								out.createNewFile();
							}
							FileOutputStream os = new FileOutputStream(out);
							a.extractFile(fh, os);
							os.close();
						} catch (Exception ex) {
							logger.error(ex.getMessage(), ex);
						}
					}
					fh = a.nextFileHeader();
				}
				a.close();
				result = true;
			}
		} catch (Exception e) {
			logger.error(e.getMessage(), e);
		}
		return result;
	}
}
